

set more off
mat drop _all




use "$path_data/data/temp/attrition_check.dta", clear








*** Test for DT
preserve
replace DT_per_min_pre = . if wrongDT == 1
replace DT_time_pre = . if wrongDT == 1
replace DT_score_pre = . if wrongDT == 1
gen missing_DT = 0
replace missing_DT = 1 if DT_per_min_pre == .
egen mean_DT_per_min_pre = mean(DT_per_min_pre)
replace DT_per_min_pre = mean_DT_per_min_pre if DT_per_min_pre == .
egen mean_DT_time_pre = mean(DT_time_pre)
replace DT_time_pre = mean_DT_time_pre if DT_time_pre == .
egen mean_DT_score_pre = mean(DT_score_pre)
replace DT_score_pre = mean_DT_score_pre if DT_score_pre == .
drop if DT_per_min_post == .






foreach j in n mean sd {
tabstat DT_per_min_pre DT_score_pre DT_time_pre DT_per_min_post DT_score_post DT_time_post if treatment == 1, stat(`j') save
matrix list r(StatTotal)
matrix `j'_tr1_DT = r(StatTotal)
}

foreach j in n mean sd {
tabstat DT_per_min_pre DT_score_pre DT_time_pre DT_per_min_post DT_score_post DT_time_post if treatment == 0, stat(`j') save
matrix list r(StatTotal)
matrix `j'_ct1_DT = r(StatTotal)
}



matrix n_dif1_DT = J(1,6,.)
forvalues i = 1/6 {
	matrix n_dif1_DT[1,`i'] = n_tr1_DT[1,`i'] + n_ct1_DT[1,`i']
}


matrix mean_dif1_DT = J(1,6,.)
forvalues i = 1/6 {
	matrix mean_dif1_DT[1,`i'] = mean_tr1_DT[1,`i'] - mean_ct1_DT[1,`i']
}


foreach dep in DT_per_min_pre DT_score_pre DT_time_pre DT_per_min_post DT_score_post DT_time_post {
	foreach s in mean se pv obs wpv {
		matrix r1_`dep'_`s' = J(1,2,.)
	}

	reg `dep' treatment, cluster(school_no)
	
	matrix r1_`dep'_temp = r(table)
	matrix r1_`dep'_r1 = e(r2)
	matrix r1_`dep'_obs = e(N)


	foreach j in 1 2 {
	* beta
	matrix r1_`dep'_mean[1,`j'] = r1_`dep'_temp[1,`j']
	* standard error
	matrix r1_`dep'_se[1,`j'] = r1_`dep'_temp[2,`j']
	* p value
	matrix r1_`dep'_pv[1,`j'] = r1_`dep'_temp[4,`j']
	}


	
	
	
	cgmwildboot_2 `dep' treatment, cluster(school_no) bootcluster(school_no) null(0) reps(999) seed(1)
	matrix m1_`dep'_p = e(Pvals)
	matrix m1_`dep'_b = e(b)
	matrix m1_`dep'_v = e(V)
}









restore









* Test for PTS




preserve
gen missing_PTS = 0
replace missing_PTS = 1 if pts_pre == .
egen mean_pts_pre = mean(pts_pre)
replace pts_pre = mean_pts_pre if pts_pre == .
drop if pts_post == .






foreach j in n mean sd {
tabstat pts_pre pts_post if treatment == 1 , stat(`j') save
matrix list r(StatTotal)
matrix `j'_tr1_PTS = r(StatTotal)
}

foreach j in n mean sd {
tabstat pts_pre pts_post if treatment == 0, stat(`j') save
matrix list r(StatTotal)
matrix `j'_ct1_PTS = r(StatTotal)
}

matrix n_dif1_PTS = J(1,6,.)
forvalues i = 1/6 {
	matrix n_dif1_PTS[1,`i'] = n_tr1_PTS[1,`i'] + n_ct1_PTS[1,`i']
}


matrix mean_dif1_PTS = J(1,6,.)
forvalues i = 1/6 {
	matrix mean_dif1_PTS[1,`i'] = mean_tr1_PTS[1,`i'] - mean_ct1_PTS[1,`i']
}



foreach dep in pts_pre pts_post {
	foreach s in mean se pv obs wpv {
		matrix r1_`dep'_`s' = J(1,2,.)
	}

	reg `dep' treatment, cluster(school_no)
	matrix r1_`dep'_temp = r(table)
	matrix r1_`dep'_r1 = e(r2)
	matrix r1_`dep'_obs = e(N)


	foreach j in 1 2 {
	* beta
	matrix r1_`dep'_mean[1,`j'] = r1_`dep'_temp[1,`j']
	* standard error
	matrix r1_`dep'_se[1,`j'] = r1_`dep'_temp[2,`j']
	* p value
	matrix r1_`dep'_pv[1,`j'] = r1_`dep'_temp[4,`j']
	}


	cgmwildboot_2 `dep' treatment, cluster(school_no) bootcluster(school_no) null(0) reps(999) seed(1)
	matrix m1_`dep'_p = e(Pvals)
	matrix m1_`dep'_b = e(b)
	matrix m1_`dep'_v = e(V)
}






restore















preserve

gen missing_rosen = 0
replace missing_rosen = 1 if rosen_pre == .
egen mean_rosen_pre = mean(rosen_pre)
replace rosen_pre = mean_rosen_pre if rosen_pre == .
drop if rosen_el == .








foreach j in n mean sd {
tabstat rosen_pre rosen_post if treatment == 1 & rosen_el != . , stat(`j') save
matrix list r(StatTotal)
matrix `j'_tr1_rosen = r(StatTotal)
}

foreach j in n mean sd {
tabstat rosen_pre rosen_post if treatment == 0 & rosen_el != ., stat(`j') save
matrix list r(StatTotal)
matrix `j'_ct1_rosen = r(StatTotal)
}

matrix n_dif1_rosen = J(1,6,.)
forvalues i = 1/6 {
	matrix n_dif1_rosen[1,`i'] = n_tr1_rosen[1,`i'] + n_ct1_rosen[1,`i']
}



matrix mean_dif1_rosen = J(1,6,.)
forvalues i = 1/6 {
	matrix mean_dif1_rosen[1,`i'] = mean_tr1_rosen[1,`i'] - mean_ct1_rosen[1,`i']
}


*foreach dep in DT_per_min_pre DT_score_pre DT_time_pre DT_per_min_post DT_score_post DT_time_post {
*	cgmwildboot_2 `dep' treatment, cluster(school_no) bootcluster(school_no) null(0) reps(999) seed(1)
*	matrix m_`dep'_p = e(Pvals)
*}

foreach dep in rosen_pre rosen_post {
	foreach s in mean se pv obs wpv {
		matrix r1_`dep'_`s' = J(1,2,.)
	}

	reg `dep' treatment if rosen_el != ., cluster(school_no)

	matrix r1_`dep'_temp = r(table)
	matrix r1_`dep'_r1 = e(r2)
	matrix r1_`dep'_obs = e(N)


	foreach j in 1 2 {
	* beta
	matrix r1_`dep'_mean[1,`j'] = r1_`dep'_temp[1,`j']
	* standard error
	matrix r1_`dep'_se[1,`j'] = r1_`dep'_temp[2,`j']
	* p value
	matrix r1_`dep'_pv[1,`j'] = r1_`dep'_temp[4,`j']
	}


	cgmwildboot_2 `dep' treatment if rosen_el != ., cluster(school_no) bootcluster(school_no) null(0) reps(999) seed(1)
	matrix m1_`dep'_p = e(Pvals)
	matrix m1_`dep'_b = e(b)
	matrix m1_`dep'_v = e(V)
}




restore




preserve

gen missing_cpcs = 0
replace missing_cpcs = 1 if cpcs_pre == .
egen mean_cpcs_pre = mean(cpcs_pre)
replace cpcs_pre = mean_cpcs_pre if rosen_pre == .
drop if cpcs_el == .




foreach j in n mean sd {
tabstat cpcs_pre cpcs_post if treatment == 1 & cpcs_el != ., stat(`j') save
matrix list r(StatTotal)
matrix `j'_tr1_cpcs = r(StatTotal)
}

foreach j in n mean sd {
tabstat cpcs_pre cpcs_post if treatment == 0 & cpcs_el != ., stat(`j') save
matrix list r(StatTotal)
matrix `j'_ct1_cpcs = r(StatTotal)
}


matrix n_dif1_cpcs = J(1,6,.)
forvalues i = 1/6 {
	matrix n_dif1_cpcs[1,`i'] = n_tr1_cpcs[1,`i'] + n_ct1_cpcs[1,`i']
}



matrix mean_dif1_cpcs = J(1,6,.)
forvalues i = 1/6 {
	matrix mean_dif1_cpcs[1,`i'] = mean_tr1_cpcs[1,`i'] - mean_ct1_cpcs[1,`i']
}


foreach dep in cpcs_pre cpcs_post {
	foreach s in mean se pv obs wpv {
		matrix r1_`dep'_`s' = J(1,2,.)
	}

	reg `dep' treatment if  cpcs_el != ., cluster(school_no)
	reg `dep' treatment if  cpcs_el != ., cluster(school_no)

	matrix r1_`dep'_temp = r(table)
	matrix r1_`dep'_r1 = e(r2)
	matrix r1_`dep'_obs = e(N)


	foreach j in 1 2 {
	* beta
	matrix r1_`dep'_mean[1,`j'] = r1_`dep'_temp[1,`j']
	* standard error
	matrix r1_`dep'_se[1,`j'] = r1_`dep'_temp[2,`j']
	* p value
	matrix r1_`dep'_pv[1,`j'] = r1_`dep'_temp[4,`j']
	}

	cgmwildboot_2 `dep' treatment if cpcs_el != ., cluster(school_no) bootcluster(school_no) null(0) reps(999) seed(1)
	matrix m1_`dep'_p = e(Pvals)
	matrix m1_`dep'_b = e(b)
	matrix m1_`dep'_v = e(V)
}


restore




preserve

drop if gender == . 
drop if (rosen_bl ==. | rosen_el == . ) & (cpcs_bl ==. | cpcs_el == . ) & ///
(pts_pre ==. | pts_post == . ) & (DT_per_min_pre ==. | DT_per_min_post == . )

foreach j in n mean sd {
tabstat gender if treatment == 1 , stat(`j') save
matrix list r(StatTotal)
matrix `j'_tr1_demo = r(StatTotal)
}

foreach j in n mean sd {
tabstat gender if treatment == 0 , stat(`j') save
matrix list r(StatTotal)
matrix `j'_ct1_demo = r(StatTotal)
}



matrix n_dif1_demo = J(1,6,.)
forvalues i = 1/6 {
	matrix n_dif1_demo[1,`i'] = n_tr1_demo[1,`i'] + n_ct1_demo[1,`i']
}


matrix mean_dif1_demo = J(1,6,.)
forvalues i = 1/6 {
	matrix mean_dif1_demo[1,`i'] = mean_tr1_demo[1,`i'] - mean_ct1_demo[1,`i']
}


foreach dep in gender {
	foreach s in mean se pv obs wpv {
		matrix r1_`dep'_`s' = J(1,2,.)
	}

	reg `dep' treatment , cluster(school_no)
	
	matrix r1_`dep'_temp = r(table)
	matrix r1_`dep'_r1 = e(r2)
	matrix r1_`dep'_obs = e(N)


	foreach j in 1 2 {
	* beta
	matrix r1_`dep'_mean[1,`j'] = r1_`dep'_temp[1,`j']
	* standard error
	matrix r1_`dep'_se[1,`j'] = r1_`dep'_temp[2,`j']
	* p value
	matrix r1_`dep'_pv[1,`j'] = r1_`dep'_temp[4,`j']
	}

	
	cgmwildboot_2 `dep' treatment , cluster(school_no) bootcluster(school_no) null(0) reps(999) seed(1)
	matrix m1_`dep'_p = e(Pvals)
	matrix m1_`dep'_b = e(b)
	matrix m1_`dep'_v = e(V)
}






restore




preserve
drop if age_pre == .
drop if (rosen_bl ==. | rosen_el == . ) & (cpcs_bl ==. | cpcs_el == . ) & ///
(pts_pre ==. | pts_post == . ) & (DT_per_min_pre ==. | DT_per_min_post == . )

foreach j in n mean sd {
tabstat age_pre age_sq_pre if treatment == 1 , stat(`j') save
matrix list r(StatTotal)
matrix `j'_tr1_demo2 = r(StatTotal)
}

foreach j in n mean sd {
tabstat age_pre age_sq_pre if treatment == 0 , stat(`j') save
matrix list r(StatTotal)
matrix `j'_ct1_demo2 = r(StatTotal)
}



matrix n_dif1_demo2 = J(1,6,.)
forvalues i = 1/6 {
	matrix n_dif1_demo2[1,`i'] = n_tr1_demo2[1,`i'] + n_ct1_demo2[1,`i']
}


matrix mean_dif1_demo2 = J(1,6,.)
forvalues i = 1/6 {
	matrix mean_dif1_demo2[1,`i'] = mean_tr1_demo2[1,`i'] - mean_ct1_demo2[1,`i']
}


foreach dep in age_pre age_sq_pre {
	foreach s in mean se pv obs wpv {
		matrix r1_`dep'_`s' = J(1,2,.)
	}

	reg `dep' treatment , cluster(school_no)
	
	matrix r1_`dep'_temp = r(table)
	matrix r1_`dep'_r1 = e(r2)
	matrix r1_`dep'_obs = e(N)


	foreach j in 1 2 {
	* beta
	matrix r1_`dep'_mean[1,`j'] = r1_`dep'_temp[1,`j']
	* standard error
	matrix r1_`dep'_se[1,`j'] = r1_`dep'_temp[2,`j']
	* p value
	matrix r1_`dep'_pv[1,`j'] = r1_`dep'_temp[4,`j']
	}

	
	cgmwildboot_2 `dep' treatment , cluster(school_no) bootcluster(school_no) null(0) reps(999) seed(1)
	matrix m1_`dep'_p = e(Pvals)
	matrix m1_`dep'_b = e(b)
	matrix m1_`dep'_v = e(V)
}


restore

















forvalues j = 1/1 {

foreach dep in DT_per_min_pre DT_score_pre DT_time_pre DT_per_min_post ///
DT_score_post DT_time_post pts_pre pts_post rosen_pre rosen_post ///
cpcs_pre cpcs_post gender age_pre age_sq_pre {


forvalues item = 1/1 {
		forvalues scale = 1/1 {
		if m`j'_`dep'_p[`item', `scale']<=0.01 {
			local s_`j'_`dep'_`item'`scale' %3s "***"
		}
		else if (m`j'_`dep'_p[`item', `scale']>0.01) & (m`j'_`dep'_p[`item', `scale']<=0.05) {
			local s_`j'_`dep'_`item'`scale' %2s "**"
		}
		else if (m`j'_`dep'_p[`item', `scale']>0.05) & (m`j'_`dep'_p[`item', `scale']<=0.10) {
			local s_`j'_`dep'_`item'`scale' %1s "*"
		}
		else {
			local s_`j'_`dep'_`item'`scale'  ""
		}
	} // forvalues `scale' loop

}  // forvalues `item' loop


} // foreach `dep' loop

} // forvalues `j' loop



tempname hh2
file open `hh2' using "$pardir/table1.tex", write replace
file write `hh2' "" _newline
file write `hh2' "% Date: `c(current_date)'" _newline
file write `hh2' "% Time: `c(current_time)'" _newline
file write `hh2' "" _newline


file write `hh2' "\begin{table}[t!]\footnotesize" _newline
file write `hh2' "  \centering" _newline
file write `hh2' "  \caption{Baseline Balance}" _newline
file write `hh2' "\label{tab:summary}" _newline
file write `hh2' "\scalebox{0.9}{" _newline
file write `hh2' "\begin{threeparttable}" _newline

file write `hh2' "\begin{tabular}{lcccc}\toprule\midrule" _newline




*file write `hh2' "\cmidrule{2-5} & &&&      \\" _newline

file write `hh2' " Dependent Variable & Treatment &  Control  & Difference & N   \\\midrule" _newline


file write `hh2' "    DT Score per min\textsuperscript{a}   & " %04.3f (mean_tr1_DT[1,1]) " &   " %04.3f (mean_ct1_DT[1,1]) " & " %04.3f (mean_dif1_DT[1,1]) `s_1_DT_per_min_pre_11' " & " (n_dif1_DT[1,1]) " \\" _newline
file write `hh2' "    									    & [" %04.3f (sd_tr1_DT[1,1]) "] &   [" %04.3f (sd_ct1_DT[1,1]) "] & (" %04.3f (r1_DT_per_min_pre_se[1,1])             ") &                     \\" _newline


file write `hh2' "    DT Score                              & " %04.3f (mean_tr1_DT[1,2]) " &   " %04.3f (mean_ct1_DT[1,2]) " & " %04.3f (mean_dif1_DT[1,2]) `s_1_DT_score_pre_11' " & " (n_dif1_DT[1,2]) " \\" _newline
file write `hh2' "    									    & [" %04.3f (sd_tr1_DT[1,2]) "] &   [" %04.3f (sd_ct1_DT[1,2]) "] & (" %04.3f (r1_DT_score_pre_se[1,1])             ") &                     \\" _newline


file write `hh2' "    DT Time                               & " %04.3f (mean_tr1_DT[1,3]) " &   " %04.3f (mean_ct1_DT[1,3]) " & " %04.3f (mean_dif1_DT[1,3]) `s_1_DT_time_pre_11' " & " (n_dif1_DT[1,3]) " \\" _newline
file write `hh2' "    									    & [" %04.3f (sd_tr1_DT[1,3]) "] &   [" %04.3f (sd_ct1_DT[1,3]) "] & (" %04.3f (r1_DT_time_pre_se[1,1])             ") &                     \\" _newline


file write `hh2' "    PTSII-C Score\textsuperscript{b}      & " %04.3f (mean_tr1_PTS[1,1]) " &   " %04.3f (mean_ct1_PTS[1,1]) " & " %04.3f (mean_dif1_PTS[1,1]) `s_1_pts_pre_11' " & " (n_dif1_PTS[1,1]) " \\" _newline
file write `hh2' "    									    & [" %04.3f (sd_tr1_PTS[1,1]) "] &   [" %04.3f (sd_ct1_PTS[1,1]) "] & (" %04.3f (r1_pts_pre_se[1,1])              ") &                      \\" _newline


file write `hh2' "    RSES Index\textsuperscript{c}               & " %04.3f (mean_tr1_rosen[1,1]) " &   " %04.3f (mean_ct1_rosen[1,1]) " & " %04.3f (r1_rosen_pre_mean[1,1]) `s_1_rosen_pre_11' " & " (n_dif1_rosen[1,1]) " \\" _newline
file write `hh2' "    									    & [" %04.3f (sd_tr1_rosen[1,1]) "] &   [" %04.3f (sd_ct1_rosen[1,1]) "] & (" %04.3f (r1_rosen_pre_se[1,1])                ") &                        \\" _newline


file write `hh2' "    CPCS Index\textsuperscript{c}               & " %04.3f (mean_tr1_cpcs[1,1]) " &   " %04.3f (mean_ct1_cpcs[1,1]) " & " %04.3f (r1_cpcs_pre_mean[1,1]) `s_1_cpcs_pre_11' " & " (n_dif1_cpcs[1,1]) " \\" _newline
file write `hh2' "    									    & [" %04.3f (sd_tr1_cpcs[1,1]) "] &   [" %04.3f (sd_ct1_cpcs[1,1]) "] & (" %04.3f (r1_cpcs_pre_se[1,1])               ") &                       \\" _newline


file write `hh2' "    Female                                & " %04.3f (mean_tr1_demo[1,1]) " &   " %04.3f (mean_ct1_demo[1,1]) " & " %04.3f (mean_dif1_demo[1,1]) `s_1_gender_11' " & " (n_dif1_demo[1,1]) " \\" _newline
file write `hh2' "    									    & [" %04.3f (sd_tr1_demo[1,1]) "] &   [" %04.3f (sd_ct1_demo[1,1]) "] & (" %04.3f (r1_gender_se[1,1])               ") &                       \\" _newline


file write `hh2' "    Age                                   & " %04.3f (mean_tr1_demo2[1,1]) " &   " %04.3f (mean_ct1_demo2[1,1]) " & " %04.3f (mean_dif1_demo2[1,1]) `s_1_age_pre_11' " & " (n_dif1_demo2[1,2]) " \\ " _newline
file write `hh2' "    									    & [" %04.3f (sd_tr1_demo2[1,1]) "] &   [" %04.3f (sd_ct1_demo2[1,1]) "] & (" %04.3f (r1_age_pre_se[1,1])               ") &                       \\" _newline


file write `hh2' "    Age Squared                            & " %04.3f (mean_tr1_demo2[1,2]) " &   " %04.3f (mean_ct1_demo2[1,2]) " & " %04.3f (mean_dif1_demo2[1,2]) `s_1_age_sq_pre_1' " & " (n_dif1_demo2[1,2]) " \\ " _newline
file write `hh2' "    									    & [" %04.3f (sd_tr1_demo2[1,2]) "] &   [" %04.3f (sd_ct1_demo2[1,2]) "] & (" %04.3f (r1_age_sq_pre_se[1,1])               ") &                       \\\bottomrule" _newline







file write `hh2' "\end{tabular}" _newline
file write `hh2' "\end{threeparttable}" _newline
file write `hh2' "}" _newline
file write `hh2' "\label{tab:addlabel}%" _newline
file write `hh2' "\end{table}" _newline

file write `hh2' "" _newline
file write `hh2' "" _newline
file write `hh2' "" _newline
file write `hh2' "" _newline

file close `hh2'












